# -*- coding: utf-8 -*-
"""
Created on Fri Mar 18 20:14:19 2016
To operate the TXT file in one column.
首先，打印出目录下面所有的文件名。
然后，将其中txt文件融合为一个文件，名字叫做merge_txt.txt

@author: pwatx1
"""
import os
import sys

file_dir = raw_input("Please input the file directory: ")
file_list = os.listdir(file_dir)
file_list.sort()
file_list2 = [i.decode('gbk') for i in file_list]
file_list3 = [i.encode('utf-8') for i in file_list2]
for name in file_list3:
    print name
print '___________________________'

os.chdir(file_dir)
#这一段的目的是为了剔除所有不是txt的文件。
for notxt_file in file_list:
    split_name = os.path.splitext(notxt_file)
    if split_name[1] != '.txt':
        file_list.remove(notxt_file)
    elif split_name[0] == 'Merge_txt':
        file_list.remove('Merge_txt.txt')
        
#首先，清空之前可能存在的Merge_txt，然后将所有的txt文件内容写入到Merge_txt中。
clear = open("Merge_txt.txt", 'w')
clear.close()
merge_file = open("Merge_txt.txt", 'a+')
for txt_file in file_list:
    txt_content = open(txt_file, 'r+')
    merge_file.write(txt_content.read())
    txt_content.close()
    
merge_file.close()
